#!/bin/bash
if [[ $1 && $2 ]]; then
	
	local=`pwd`/local
		
	mkdir -p data data/local data/$1 data/$2

	echo "Preparing train and test data"
	echo "make wav.scp for $1 $2"

	rm -rf data/mfcc data/log
	
	cd asr_swahili/data
	
        pushd $1
	cp wav.scp ../../../data/$1/wav.scp
	popd

	pushd $2/wav5
	ls */*.wav |  sed 's/^/asr_swahili\/data\/test\/wav5\//g' > tutu1
	cat tutu1 | sed "s/\//#/g" | awk 'BEGIN{FS="#"} {print $6}' | sed "s/\.wav//g" > tutu2
	paste tutu2 tutu1 > ../../../../data/$2/wav.scp
	rm tutu1 tutu2
	popd
	
	echo "copy spk2utt, utt2spk, text for $1 $2"

	for x in $1 $2; do
	  cp $x/spk2utt ../../data/$x/.
	  cp $x/utt2spk ../../data/$x/.
	  cp $x/text ../../data/$x/.
	done
	
	pushd ../../data/local
	if [ ! -f  "swahili.arpa" ]; then
		cd ../../asr_swahili/LM
		unzip swahili.arpa.zip -d ../../data/local/
	fi
	popd		
	
	echo "Preparing data OK."
	
	cd ../..
else
        echo "ERROR: Preparing train and test data failed !"
        echo "You must have forgotten to precise train test directories"
        echo "Usage: ./prepare_data.sh train test"
fi
